import base64
import json

import pickle
from django.shortcuts import render
from django import http
from django.views import View
from django_redis import get_redis_connection

from apps.goods.models import SKU
from utils.response_code import RETCODE

"""
需求：
    1.未登录用户能实现购物车的添加，也能够保存数据
        登录用户
    2.未登录用户保存在  浏览器(cookie中)
        登录用户保存在  服务器 数据库中(redis(课程设计，真实实现))
    3.未登录用户保存 商品id 商品数量 勾选状态
        cookie
        key:value
        carts: { sku_id: {count:xxx,selected:True}}
        carts: {
                1:{count:5,selected:True}
                3:{count:1,selected:False}
                }
        登录用户保存 用户id 商品id 商品数量 勾选状态
          数据库 user_id,sku_id,count,selected

                     1      1       5   1
                     1      3       15  0
                     333    1       1   1
                     333    2       1   0
         Redis(课程设计,真实实现)
            Redis是保存在内存中,我们的原则是:尽量少的占用内存空间 (先实现功能)

                 user_id,sku_id,count,selected



            Hash
                user_id:
                        sku_id:count,
                        sku_id:count,

                1:
                  1:10


                  2:20


                2:
                  1:10


                  2:20
            Set
                user_id: [sku_id,sku_id] 选中的商品的id

                1:  [1]

                2:
    4. 对cookie数据进行加密处理

        1G=1024MB
        1M=1024KB
        1KB=1024B
        1B=8bytes

        1字节=8b
        b: 0/1

        A       0100    0001

        A   A   A
        0100    0001    0100    0001    0100    0001

        X   y z s
        010000      010100    000101    000001

"""


class CartsView(View):
    # 购物车的添加
    def post(self, request):
        """
         需求：
             1.当用户点击加入购物车的时候，需要让前端收集 sku_id,count,selected(可以不提交 默认是True)
             因为请求的时候会携带用户信息(如果用户登录)
         后端：
             1.后端接收数据
             2.验证数据
             3.判断用户登录状态
             4.登录用户保存在redis(连接redis  hash  set  返回)
             5.未登录用户保存在cookie中(组织数据 加密 设置cookie 返回响应)
         """
        # 1.后端接收数据
        data = json.loads(request.body.decode())
        sku_id = data.get("sku_id")
        count = data.get("count")
        selected = data.get("selected", True)
        # 2.验证数据
        # 商品id和数量必须传递
        if not all([sku_id, count]):
            return http.JsonResponse({"code": RETCODE.PARAMERR, "errmsg": "参数不全！"})
        # 判断商品是否存在
        try:
            sku = SKU.objects.get(pk=sku_id)
        except SKU.DoesNotExist:
            return http.JsonResponse({"code": RETCODE.NODATAERR, "errmsg": "没有此商品！"})
        # 判断商品的个数是否为整型
        try:
            count = int(count)
        except Exception as e:
            return http.JsonResponse({"code": RETCODE.PARAMERR, "errmsg": "参数错误！"})
        # 判断选中状态是否为bool值
        if not (selected, bool):
            return http.JsonResponse({"code": RETCODE.PARAMERR, "errmsg": "参数错误！"})
        # 3.判断用户登录状态
        user = request.user
        if user.is_authenticated:
            # 4.登录用户保存在redis(连接redis  hash  set  返回)
            redis_conn = get_redis_connection("carts")
            redis_conn.hset("carts_%s" % user.id, sku_id, count)
            redis_conn.sadd("selected_%s" % user.id, sku_id)
            return http.JsonResponse({"code": RETCODE.OK, "errmsg": "ok"})
        else:
            # 5.未登录用户保存在cookie中(组织数据 加密 设置cookie 返回响应)
            # 先读取cookie数据，判断cookie数据中
            carts = request.COOKIES.get("carts")
            if carts is not None:
                # 如果存在，则需要先对数据进行解密
                # 先将base64数据解码为bytes
                decode = base64.b64decode(carts)
                # 再将bytes类型的数据转换为字典
                cookie_cart = pickle.loads(decode)
            else:
                # 如果不存在，则初始化字典
                cookie_cart = {}
            # 判断sku_id 是否存在于cookie_cart 在则累加
            if sku_id in cookie_cart:
                # 先把原数据获取到
                orginal_count = cookie_cart[sku_id]['count']
                # 再累加
                count += orginal_count

            # 再更新数据
            cookie_cart[sku_id] = {
                'count': count,
                'selected': selected
            }
            # 加密
        dumps = pickle.dumps(cookie_cart)
        cookie_data = base64.b64encode(dumps)
        response = http.JsonResponse({"code": RETCODE.OK, "errmsg": "ok"})
        response.set_cookie("carts", cookie_data.decode(), max_age=3600)
        print(cookie_data.decode())
        return response

    # 购物车的展示
    def get(self, request):
        # 判断用户是否登录
        user = request.user
        if user.is_authenticated:
            # 登录用户到redis中获取数据
            redis_conn = get_redis_connection("carts")
            # pl = redis_conn.pipeline()
            # pl.hgetall('carts_%s'%user.id)
            # pl.smembers('selected_%s'%user.id)
            # pl.execute()
            sku_id_count = redis_conn.hgetall('carts_%s' % user.id)
            selected_ids = redis_conn.smembers('selected_%s' % user.id)

            # 因为 redis和cookie 的数据格式不一致，我们需要转换数据
            # 　将redis转换为cookie的格式
            # carts:{sku_id{count:xxx,selected:xxx}}
            cookie_cart = {}
            for sku_id, count in sku_id_count.items():
                if sku_id in selected_ids:
                    selected = True
                else:
                    selected = False
                cookie_cart[int(sku_id)] = {
                    "count": int(count),
                    "selected": selected
                    # "selecte":sku_id in selected_ids
                }
        else:
            # 未登录用户到cookie中获取数据
            carts = request.COOKIES.get("carts")
            if carts is not None:
                decode = base64.b64decode(carts)
                cookie_cart = pickle.loads(decode)
            else:
                # 没有数据　初始化
                cookie_cart = {}

        # 获取所有商品id
        ids = cookie_cart.keys()
        # 根据商品id获取商品的详细信息
        skus = []
        for id in ids:
            sku = SKU.objects.get(pk=id)
            skus.append({
                "id": sku.id,
                "name": sku.name,
                "count": cookie_cart.get(sku.id).get('count'),
                "selected": str(cookie_cart.get(sku.id).get('selected')),
                "default_image_url": sku.default_image.url,
                "price": str(sku.price),
                "amount": str(sku.price * cookie_cart.get(sku.id).get('count'))
            })
        context = {
            'cart_skus': skus
        }
        return render(request, 'cart.html', context=context)

    # 购物车的修改
    def put(self, request):
        """
        1.接受数据
        2.验证数据
            2.1 sku_id,selected,count 都要提交过来
            2.2 验证商品id
            2.3 验证商品数量
            2.4 验证选中状态是否为bool
        3.获取用户信息，根据用户信息判断
        4.登录用户redis
            4.1 连接redis
            4.2 更新数据
            4.3 返回响应
        5.未登录用户cookie
            5.1 获取cookie中的carts数据
            5.2 判断数据是否存在
            5.3 更新数据
            5.4 对数据进行加密处理
            5.5 返回响应
        """
        # 1.接受数据
        data = json.loads(request.body.decode())
        sku_id = data.get("sku_id")
        count = data.get("count")
        selected = data.get("selected", True)
        # 2.验证数据
        #     2.1 sku_id,selected,count 都要提交过来
        if not all([sku_id, count, selected]):
            return http.JsonResponse({'code': RETCODE.PARAMERR, 'errmsg': '参数不全'})
        # 2.2 验证商品id
        try:
            sku = SKU.objects.get(pk=sku_id)
        except SKU.DoesNotExist:
            return http.JsonResponse({'code': RETCODE.PARAMERR, 'errmsg': '没有此商品'})
        # 2.3 验证商品数量
        try:
            count = int(count)
        except Exception as e:
            return http.JsonResponse({'code': RETCODE.PARAMERR, 'errmsg': '参数错误'})
        # 2.4 验证选中状态是否为bool
        if not isinstance(selected, bool):
            return http.JsonResponse({'code': RETCODE.PARAMERR, 'errmsg': '参数错误'})
        # 3.获取用户信息，根据用户信息判断
        user = request.user
        if user.is_authenticated:
            # 4.登录用户redis
            #     4.1 连接redis
            redis_conn = get_redis_connection("carts")
            #     4.2 更新数据
            redis_conn.hset('carts_%s' % user.id, sku_id, count)
            if selected:
                redis_conn.sadd('selected_%s' % user.id, sku_id)
            else:
                redis_conn.srem('selected_%s' % user.id, sku_id)
            # 4.3 返回响应
            cart_sku = {
                'id': sku.id,
                'count': count,
                'selected': selected,
                'name': sku.name,
                'default_image_url': sku.default_image.url,
                'price': sku.price,
                'amount': sku.price * count
            }
            return http.JsonResponse({"code": RETCODE.OK, "errmsg": "ok", "cart_sku": cart_sku})
        else:
            # 5.未登录用户cookie
            #     5.1 获取cookie中的carts数据
            carts = request.COOKIES.get("carts")
            #     5.2 判断数据是否存在
            if carts is not None:
                # 存在 解密
                cookie_cart = pickle.loads(base64.b64decode(carts))
            else:
                # 不存在
                cookie_cart = {}
            # 5.3 更新数据
            if sku_id in cookie_cart:
                cookie_cart[sku_id] = {
                    'count': count,
                    'selected': selected
                }
                #     5.4 对数据进行加密处理
                cookie_data = base64.b64encode(pickle.dumps(cookie_cart))
            # 5.5 返回响应
            cart_sku = {
                'id': sku.id,
                'count': count,
                'selected': selected,
                'name': sku.name,
                'default_image_url': sku.default_image.url,
                'price': sku.price,
                'amount': sku.price * count
            }
            response = http.JsonResponse({"code": RETCODE.OK, "errmsg": "ok", "cart_sku": cart_sku})
            response.set_cookie('carts', cookie_data.decode(), 3600)
            return response

    # 购物车的删除
    def delete(self, request):
        """
        1.接受数据(sku_id)
        2.验证数据(验证商品是否存在)
        3.获取用户信息
        4.登录用户操作redis
            4.1 连接redis
            4.2 删除数据
            4.3 返回响应
        5.未登录用户操作cookie
            5.1 获取cookie信息
            5.2 判断数据是否存在
            5.3 删除数据
            5.4 对最新的数据进行加密处理
            5.5 返回响应
        """
        # 1.接受数据(sku_id)
        data = json.loads(request.body.decode())
        sku_id = data.get("sku_id")
        # 2.验证数据(验证商品是否存在)
        try:
            sku = SKU.objects.get(pk=sku_id)
        except SKU.DoesNotExist:
            return http.JsonResponse({"code": RETCODE.NODATAERR, "errmsg": "没有数据"})
        # 3.获取用户信息
        user = request.user
        if user.is_authenticated:
            # 4.登录用户操作redis
            #     4.1 连接redis
            redis_conn = get_redis_connection("carts")
            #     4.2 删除数据
            # redis_conn.hdel("carts_%s"%user.id,sku_id)
            # redis_conn.srem("carts_%s"%user.id,sku_id)
            pl = redis_conn.pipeline()
            # 删除键，就等价于删除了整条记录
            pl.hdel('carts_%s' % user.id, sku_id)
            pl.srem('selected_%s' % user.id, sku_id)
            pl.execute()
            #     4.3 返回响应
            return http.JsonResponse({"code": RETCODE.OK, "errmsg": "ok"})
        else:
            # 5.未登录用户操作cookie
            #     5.1 获取cookie信息
            carts = request.COOKIES.get("carts")
            #     5.2 判断数据是否存在
            if carts is not None:
                cookie_cart = pickle.loads(base64.b64decode(carts))
            else:
                cookie_cart = {}
            # 5.3 删除数据
            if sku_id in cookie_cart:
                del cookie_cart[sku_id]
            # 5.4 对最新的数据进行加密处理
            cookie_data = base64.b64encode(pickle.dumps(cookie_cart))
            #     5.5 返回响应
            response = http.JsonResponse({"code": RETCODE.OK, "errmsg": "ok"})
            response.set_cookie("carts", cookie_data, 3600)
            return response
